home *** CD-ROM | disk | FTP | other *** search
/ Mac Easy 2010 May / Mac Life Ubuntu.iso / casper / filesystem.squashfs / usr / share / pyshared / DistUpgrade / DistUpgradeMain.py < prev    next >
Encoding:
Python Source  |  2009-04-27  |  5.0 KB  |  127 lines
  1. # DistUpgradeMain.py 
  2. #  
  3. #  Copyright (c) 2004-2008 Canonical
  4. #  
  5. #  Author: Michael Vogt <michael.vogt@ubuntu.com>
  6. # 
  7. #  This program is free software; you can redistribute it and/or 
  8. #  modify it under the terms of the GNU General Public License as 
  9. #  published by the Free Software Foundation; either version 2 of the
  10. #  License, or (at your option) any later version.
  11. # 
  12. #  This program is distributed in the hope that it will be useful,
  13. #  but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. #  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15. #  GNU General Public License for more details.
  16. # 
  17. #  You should have received a copy of the GNU General Public License
  18. #  along with this program; if not, write to the Free Software
  19. #  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  20. #  USA
  21.  
  22. from DistUpgradeController import DistUpgradeController
  23. from DistUpgradeConfigParser import DistUpgradeConfig
  24. import logging
  25. import os
  26. import sys
  27. import glob
  28. import shutil
  29. from datetime import datetime
  30. from optparse import OptionParser
  31. from gettext import gettext as _
  32.  
  33. def do_commandline():
  34.     " setup option parser and parse the commandline "
  35.     parser = OptionParser()
  36.     parser.add_option("-s", "--sandbox", dest="useAufs", default=False,
  37.                       action="store_true",
  38.                       help=_("Sandbox upgrade using aufs"))
  39.     parser.add_option("-c", "--cdrom", dest="cdromPath", default=None,
  40.                       help=_("Use the given path to search for a cdrom with upgradable packages"))
  41.     parser.add_option("--have-prerequists", dest="havePrerequists",
  42.                       action="store_true", default=False)
  43.     parser.add_option("--with-network", dest="withNetwork",action="store_true")
  44.     parser.add_option("--without-network", dest="withNetwork",action="store_false")
  45.     parser.add_option("--frontend", dest="frontend",default=None,
  46.                       help=_("Use frontend. Currently available: \n"\
  47.                              "DistUpgradeViewText, DistUpgradeViewGtk, DistUpgradeViewKDE"))
  48.     parser.add_option("--mode", dest="mode",default="desktop",
  49.                       help=_("*DEPRECATED* this option will be ignore"))
  50.     parser.add_option("--partial", dest="partial", default=False,
  51.                       action="store_true", 
  52.                       help=_("Perform a partial upgrade only (no sources.list rewriting)"))
  53.     parser.add_option("--datadir", dest="datadir", default=None,
  54.                       help=_("Set datadir"))
  55.     return parser.parse_args()
  56.     
  57. def setup_logging(options, config):
  58.     " setup the logging "
  59.     logdir = config.getWithDefault("Files","LogDir","/var/log/dist-upgrade/")
  60.     if not os.path.exists(logdir):
  61.         os.mkdir(logdir)
  62.     # check if logs exists and move logs into place
  63.     if glob.glob(logdir+"/*.log"):
  64.         now = datetime.now()
  65.         backup_dir = logdir+"/%04i%02i%02i-%02i%02i" % (now.year,now.month,now.day,now.hour,now.minute)
  66.         if not os.path.exists(backup_dir):
  67.             os.mkdir(backup_dir)
  68.         for f in glob.glob(logdir+"/*.log"):
  69.             shutil.move(f, os.path.join(backup_dir,os.path.basename(f)))
  70.     fname = os.path.join(logdir,"main.log")
  71.     # do not overwrite the default main.log
  72.     if options.partial:
  73.         fname += ".partial"
  74.     logging.basicConfig(level=logging.DEBUG,
  75.                         filename=fname,
  76.                         format='%(asctime)s %(levelname)s %(message)s',
  77.                         filemode='w')
  78.     # log what config files are in use here to detect user
  79.     # changes
  80.     logging.info("Using config files '%s'" % config.config_files)
  81.     return logdir
  82.     
  83. def setup_view(options, config, logdir):
  84.     " setup view based on the config and commandline "
  85.  
  86.     # the commandline overwrites the configfile
  87.     for requested_view in [options.frontend]+config.getlist("View","View"):
  88.         if not requested_view:
  89.             continue
  90.         try:
  91.             view_modul = __import__(requested_view)
  92.             view_class = getattr(view_modul, requested_view)
  93.             break
  94.         except (ImportError, AttributeError, TypeError), e:
  95.             logging.warning("can't import view '%s' (%s)" % (requested_view,e))
  96.             print "can't load %s (%s)" % (requested_view, e)
  97.     else:
  98.         logging.error("No view can be imported, aborting")
  99.         print "No view can be imported, aborting"
  100.         sys.exit(1)
  101.     return view_class(logdir=logdir)
  102.  
  103. def main():
  104.     " main method "
  105.     
  106.     # commandline setup and config
  107.     (options, args) = do_commandline()
  108.     config = DistUpgradeConfig(".")
  109.     logdir = setup_logging(options, config)
  110.  
  111.     from DistUpgradeVersion import VERSION
  112.     logging.info("release-upgrader version '%s' started" % VERSION)
  113.  
  114.     # create view and app objects
  115.     view = setup_view(options, config, logdir)
  116.     app = DistUpgradeController(view, options, datadir=options.datadir)
  117.  
  118.     # partial upgrade only
  119.     if options.partial:
  120.         if not app.doPartialUpgrade():
  121.             sys.exit(1)
  122.         sys.exit(0)
  123.  
  124.     # full upgrade
  125.     app.run()
  126.  
  127.